from openai import OpenAI  # 导入SDK客户端  


# 配置认证和URL
API_KEY = "sk-zO8exlBicZh7nJeZn5GuC5X9SPuVrZzXoGyOW0i9BFvN62ON"
BASE_URL = "https://api.chatfire.cn/v1"

# 创建客户端
client = OpenAI(
    api_key=API_KEY,
    base_url=BASE_URL
)


# 2. 构造对话消息
messages = [  
    {  
        "role": "system",  
        "content": "你是一名系统设计师，严格按要求输出完整、结构化的系统模型方案。"  
    },  
    {  
        "role": "user",  
        "content": (
            "设计一个覆盖用户端与管理后台的共享单车系统，包含用户注册/登录、实名认证、押金管理等基础功能。用户可扫码解锁车辆，系统支持异常处理；行程中提供实时数据、暂停用车等功能，"
            "结束后自动计算费用并支持多种支付方式及发票申请。会员服务提供月卡/季卡套餐及专属权益。管理后台需实现车辆投放调度、故障维修处理等功能。系统应满足高并发处理、定位精度及数据安全等非功能需求，"
            "同时设计数据库表结构及关键API接口，并给出异常处理方案。" ) 
    }  
]  

# 3. 调用 Chat Completions API  
try:  
    response = client.chat.completions.create(  
        model="gpt-4o",  # 模型选择
        messages=messages,  
        max_tokens=3000,  # 限制输出长度，确保复杂方案完整  
        temperature=0.6   # 控制创意度（0→严谨，2→开放）  
    )  
    # 4. 提取并打印结果  
    print("=== 共享单车系统设计方案 ===")  
    print(response.choices[0].message.content)  
except Exception as e:  
    print(f"调用失败：{e}")  # 捕获网络、权限等异常  
